package org.tea.crawl.biz.domain.repository.impl;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import org.springframework.stereotype.Repository;
import org.tea.crawl.biz.domain.entity.TeaItemUrl;
import org.tea.crawl.biz.domain.enums.ParseState;
import org.tea.crawl.biz.domain.repository.TeaItemUrlRepository;
import org.tea.crawl.biz.mapper.TeaItemUrlMapper;

import javax.annotation.Resource;
import java.util.List;

/**
 * @author lxh
 */
@Repository
public class TeaItemUrlRepositoryImpl
        implements TeaItemUrlRepository {

    @Resource
    private TeaItemUrlMapper teaItemUrlMapper;

    @Override
    public List<TeaItemUrl> findUnParsedItems() {
        return teaItemUrlMapper.selectList(Wrappers.query(TeaItemUrl.class)
                .eq("parse_state", ParseState.EMPTY.code));
    }

    @Override
    public TeaItemUrl findById(Long id) {
        return teaItemUrlMapper.selectById(id);
    }

    @Override
    public int save(TeaItemUrl teaItemUrl) {
        return teaItemUrlMapper.updateById(teaItemUrl);
    }
}
